-
Notifications
You must be signed in to change notification settings - Fork 9.4k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
refactor onnx export of two stage #5205
Conversation
Codecov Report
@@ Coverage Diff @@
## master #5205 +/- ##
==========================================
- Coverage 65.39% 65.35% -0.05%
==========================================
Files 276 276
Lines 21229 21204 -25
Branches 3561 3522 -39
==========================================
- Hits 13883 13857 -26
- Misses 6572 6599 +27
+ Partials 774 748 -26
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
Why is there a 13.7 MB tmp.onnx in this PR? |
Motivation
The recent ONNX-related development is too quickly, which made the code hard to read, and there is a problem that the return type of the same function such as
_get_bboxes
in theDenseHead
andBboxHead
would be different when export to ONNX , which will greatly affect the future development of other instance task and #4912.Modification
This PR moves all ONNX related code of two-stage model to a new function
onnx_export
in corresponding class.BC-breaking (Optional)
None